# 基础镜像，选择带有 GPU 支持的 PyTorch 镜像
FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04

# 设置工作目录
WORKDIR /app

# 安装依赖项
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3-pip \
    ffmpeg \
    && apt-get clean

# 复制 requirements.txt 文件并安装 Python 依赖项
COPY requirements.txt /app/
RUN pip3 install --no-cache-dir -r requirements.txt

# 复制应用代码
COPY . /app

# 设置默认启动命令
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "5000"]